#pragma once

#include <iostream>
#include <unistd.h>
#include <sys/wait.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#include "../comm/util.hpp"
#include "../comm/log.hpp"

// compiler.hpp负责进行代码的编译

namespace ns_compiler
{
    // 引入路径拼接功能
    using namespace ns_util;
    using namespace ns_log;

    class Compiler
    {
    public:
        Compiler()
        {}
        ~Compiler()
        {}
        //返回值：编译成功：true，否则：false
        //输入参数：待编译的文件的文件名
        //file_name: 1234 //假设用户提交的代码文件的文件名是1234
        //1234 -> ./temp/1234.cpp //由于我们的项目是针对C++代码，所以用户提交的代码的文件名的后缀是.cpp
        //1234 -> ./temp/1234.exe //编译成功要生成可执行文件，这里我们用.exe来标识
        //1234 -> ./temp/1234.stderr //编译失败时生成保存错误信息的文件
        static bool Compile(const std::string &file_name)
        {
            pid_t pid = fork();
            if(pid < 0)
            {
                LOG(ERROR) << "内部错误，创建子进程失败" << "\n";
                return false;
            }
            else if (pid == 0)
            {
                umask(0);
                //打开该文件的对应的标准错误文件，用于保存编译出错的信息，
                //O_CREAT：如果文件不存在，则创建它,O_WRONLY：以只写方式打开文件。如果文件已存在，文件指针将指向文件开头，写入时会覆盖原有内容。
                int _stderr = open(PathUtil::CompilerError(file_name).c_str(), O_CREAT | O_WRONLY, 0644);
                if(_stderr < 0){
                    LOG(WARNING) << "没有成功形成stderr文件" << "\n";
                    exit(1);
                }
                //重定向标准错误到_stderr，用于编译出错时保存编译错误的原因
                dup2(_stderr, 2);
                
                //程序替换，文件描述符表属于进程的内核数据结构（如files_struct），而程序替换操作主要替换的是进程的代码段和数据段，并不会影响内核数据结构
                //子进程: 调用编译器，完成对代码的编译工作
                //g++ -o target src -std=c++11
                execlp("g++", "g++", "-o", PathUtil::Exe(file_name).c_str(),\
                PathUtil::Src(file_name).c_str(), "-D", "COMPILER_ONLINE","-std=c++11",  nullptr/*不要忘记*/);
                LOG(ERROR) << "启动编译器g++失败，可能是参数错误" << "\n";
                exit(2);
            }
            else{
                //等待子进程完成编译服务
                waitpid(pid, nullptr, 0);
                //编译是否成功,就看有没有形成对应的可执行程序
                if(FileUtil::IsFileExists(PathUtil::Exe(file_name))){ //Exe(file_name)返回可执行程序的文件名
                    LOG(INFO) << PathUtil::Src(file_name) << " 编译成功!" << "\n";
                    return true;
                }
            }
            LOG(ERROR) << "编译失败，没有形成可执行程序" << "\n";
            return false;
        }
    };
}